<template>
   <div class="app-container">
      <el-form :model="queryParams" ref="queryRef" :inline="true" v-show="showSearch">
         <el-form-item label="训练任务" prop="name">
            <el-input
               v-model="queryParams.name"
               placeholder="请输入训练任务名称/代码"
               clearable
               style="width: 200px"
               @keyup.enter="handleQuery"
            />
         </el-form-item>
          <el-form-item label="模型" prop="model_id">
              <el-select v-model="queryParams.model_id" placeholder="模型" clearable style="width: 200px">
                  <el-option
                      v-for="dict in modelList"
                      :key="dict.id"
                      :label="dict.name"
                      :value="dict.id"
                  />
              </el-select>
          </el-form-item>
         <el-form-item label="状态" prop="status">
            <el-select v-model="queryParams.status" placeholder="状态" clearable style="width: 200px">
               <el-option
                  v-for="dict in sys_status"
                  :key="dict.value"
                  :label="dict.label"
                  :value="dict.value"
               />
            </el-select>
         </el-form-item>
         <el-form-item>
            <el-button type="primary" icon="Search" @click="handleQuery">搜索</el-button>
            <el-button icon="Refresh" @click="resetQuery">重置</el-button>
         </el-form-item>
      </el-form>

      <el-row :gutter="10" class="mb8">
         <el-col :span="1.5">
            <el-button
               type="primary"
               plain
               icon="Plus"
               @click="handleAdd"
            >新增</el-button>
         </el-col>
         <el-col :span="1.5">
            <el-button
               type="danger"
               plain
               icon="Delete"
               :disabled="multiple"
               @click="handleDelete"
            >删除</el-button>
         </el-col>
         <right-toolbar v-model:showSearch="showSearch" @queryTable="getList"></right-toolbar>
      </el-row>

      <el-table v-loading="loading" :data="trainingList" @selection-change="handleSelectionChange">
         <el-table-column type="selection" width="55" align="center" />
         <el-table-column label="任务代码" align="center" prop="code" width="100" />
         <el-table-column label="任务名称" align="center" prop="name" />  
         <el-table-column label="数据集" align="center" prop="ds_name" width="100" />
         <el-table-column label="模型" align="center" prop="model_name" width="100" />
         <el-table-column label="状态" align="center" prop="status" width="100">
            <template #default="scope">
               <dict-tag :options="sys_status" :value="scope.row.status" />
            </template>
         </el-table-column>
         <el-table-column label="开始时间" align="center" prop="start_time" />
         <el-table-column label="结束时间" align="center" prop="end_time" />
         <el-table-column label="操作" align="center" class-name="small-padding fixed-width">
            <template #default="scope">
               <el-button link type="primary" icon="Edit" @click="handleUpdate(scope.row)">修改</el-button>
               <el-button link type="danger" icon="Delete" @click="handleDelete(scope.row)">删除</el-button>
               <el-button link type="success" icon="Run" @click="handleRun(scope.row)" v-if="scope.row.status == '0' || scope.row.status == '4'">启动</el-button>
               <el-button link type="info" icon="Pause" @click="handleStop(scope.row)" v-if="scope.row.status == '1'">停止</el-button>
            </template>
         </el-table-column>
      </el-table>

      <pagination
         v-show="total > 0"
         :total="total"
         v-model:page="queryParams.page"
         v-model:limit="queryParams.limit"
         @pagination="getList"
      />

      <!-- 添加或修改公告对话框 -->
      <el-dialog :title="title" v-model="open" width="600px" append-to-body>
         <el-form ref="formRef" :model="form" :rules="rules" label-width="120px">
            <el-form-item label="任务代码" prop="code">
                <el-input v-model="form.code" placeholder="请输入训练任务代码" />
            </el-form-item>
            <el-form-item label="任务名称" prop="name">
                <el-input v-model="form.name" placeholder="请输入训练任务名称" />
            </el-form-item>
            <el-form-item label="数据集" prop="ds_id">
                <el-select v-model="form.ds_id" placeholder="数据集" clearable style="width: 200px">
                    <el-option
                        v-for="dict in dsList"
                        :key="dict.id"
                        :label="dict.name"
                        :value="dict.id"
                    />
                </el-select>
            </el-form-item>
            <el-form-item label="模型" prop="model_id">
                <el-select v-model="form.model_id" placeholder="模型" clearable style="width: 200px" @change="handleModelChange">
                    <el-option
                        v-for="dict in modelList"
                        :key="dict.id"
                        :label="dict.name"
                        :value="dict.id"
                    />
                </el-select>
            </el-form-item>
            <el-form-item label="基础训练" prop="last_id">
                <el-select v-model="form.last_id" placeholder="基础训练" clearable style="width: 200px">
                    <el-option
                        v-for="dict in trainingOptions"
                        :key="dict.id"
                        :label="dict.name"
                        :value="dict.id"
                    />
                </el-select>
            </el-form-item>
            <el-form-item label="状态" prop="status">
               <el-radio-group v-model="form.status" placeholder="状态">
                  <el-radio
                     v-for="dict in sys_status"
                     :key="dict.value"
                     :label="dict.label"
                     :value="dict.value"
                  />
               </el-radio-group>
            </el-form-item>
         </el-form>
         <template #footer>
            <div class="dialog-footer">
               <el-button type="primary" @click="submitForm">确 定</el-button>
               <el-button @click="cancel">取 消</el-button>
            </div>
         </template>
      </el-dialog>
   </div>
</template>

<script setup name="Training">
import { listTraining, getTraining, delTraining, addTraining, updateTraining, startTraining, stopTraining } from "@/api/training";
import { listDataset } from "@/api/dataset";
import { listModels } from "@/api/models";

const { proxy } = getCurrentInstance();
const sys_status = [{value:'pending', label:'暂停'}, {value:'training', label:'训练中'}, {value:'completed', label:'训练完成'}, {value:'failed', label:'失败'}, {value:'cancelled', label:'取消'}];
const trainingList = ref([]);
const trainingOptions = ref([]);
const dsList = ref([]);
const modelList = ref([]);

const open = ref(false);
const loading = ref(true);
const showSearch = ref(true);
const ids = ref([]);
const single = ref(true);
const multiple = ref(true);
const total = ref(0);
const title = ref("");

const data = reactive({
  form: {},
  queryParams: {
    page: 1,
    limit: 10,
    name: undefined,
    code: undefined,
    status: undefined
  },
  rules: {
    name: [{ required: true, message: "采集任务名称不能为空", trigger: "blur" }],
    code: [{ required: true, message: "采集任务代码不能为空", trigger: "blur" }],
    status: [{ required: true, message: "状态不能为空", trigger: "change" }],
    model_id: [{ required: true, message: "模型不能为空", trigger: "change" }],
    ds_id: [{ required: true, message: "数据集不能为空", trigger: "change" }],
  },
});

const { queryParams, form, rules } = toRefs(data);

/** 查询采集任务列表 */
function getList() {
  loading.value = true;
  listTraining(queryParams.value).then(response => {
    trainingList.value = response.rows;
    total.value = response.total;
    loading.value = false;
  });
}
/** 取消按钮 */
function cancel() {
  open.value = false;
  reset();
}
/** 表单重置 */
function reset() {
  form.value = {
    code: undefined,
    name: undefined,
    camera_id: undefined,
    ds_id: undefined,
    model_id: undefined, 
    last_id: undefined,
    status: "pending"
  };
  proxy.resetForm("formRef");
}
/** 搜索按钮操作 */
function handleQuery() {
  queryParams.value.pageNum = 1;
  getList();
}
/** 重置按钮操作 */
function resetQuery() {
  proxy.resetForm("queryRef");
  handleQuery();
}
/** 多选框选中数据 */
function handleSelectionChange(selection) {
  ids.value = selection.map(item => item.id);
  single.value = selection.length != 1;
  multiple.value = !selection.length;
}

function handleModelChange() {
  listTraining({model_id: form.value.model_id, status:'completed', page:1, limit:100}).then(response => {
    trainingOptions.value = response.rows;
  });
}
/** 新增按钮操作 */
function handleAdd() {
  reset();
  open.value = true;
  title.value = "添加训练任务";
}
/**修改按钮操作 */
function handleUpdate(row) {
  reset();
  const trainingId = row.id || ids.value;
  getTraining(trainingId).then(response => {
    form.value = response.data;
    open.value = true;
    title.value = "修改训练任务";
  });
}
/** 提交按钮 */
function submitForm() {
  proxy.$refs["formRef"].validate(valid => {
    if (valid) {
      if (form.value.id != undefined) {
        updateTraining(form.value).then(response => {
          proxy.$modal.msgSuccess("修改成功");
          open.value = false;
          getList();
        });
      } else {
        addTraining(form.value).then(response => {
          proxy.$modal.msgSuccess("新增成功");
          open.value = false;
          getList();
        });
      }
    }
  });
}
/** 删除按钮操作 */
function handleDelete(row) {
  const trainingId = row.id || ids.value
  proxy.$modal.confirm('是否确认删除训练任务编号为"' + trainingId + '"的训练任务？').then(function() {
    return delTraining(trainingId);
  }).then(() => {
    getList();
    proxy.$modal.msgSuccess("删除成功");
  }).catch(() => {});
}
/** 启动按钮操作 */
function handleRun(row) {
  const trainingId = row.id || ids.value
  proxy.$modal.confirm('是否确认启动训练任务编号为"' + trainingId + '"的训练任务？').then(function() {
    return startTraining(trainingId);
  }).then(() => {
    getList();
    proxy.$modal.msgSuccess("启动成功");
  }).catch(() => {});
}

/** 暂停按钮操作 */
function handleStop(row) {
  const trainingId = row.id || ids.value  
  proxy.$modal.confirm('是否确认停止训练任务编号为"' + trainingId + '"的训练任务？').then(function() {
    return stopTraining(trainingId);
  }).then(() => {
    getList();
    proxy.$modal.msgSuccess("暂停成功");
  }).catch(() => {});
}


getList();

listDataset({status:'3'}).then(response => {
   dsList.value = response.rows;
});
listModels().then(response => {
   modelList.value = response.rows;
});


</script>
